python square spiral